Python write into a file